package com.wyl.niupi.framework.util.informationTransfer;

import com.google.zxing.*;
import com.google.zxing.client.j2se.BufferedImageLuminanceSource;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.common.HybridBinarizer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import com.wyl.niupi.framework.util.jdkExt._PropertyUtil;

import javax.imageio.ImageIO;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

/**
 * 二维码工具类. 二维码的原理：将信息转换成二进制来表示, 然后二进制字符串映射着boolean[][], boolean[][]又映射着二维码 <br/>
 * 二维码用zxing生成比较好，参考博客：https://blog.csdn.net/lgc592519828/article/details/88805976 <br/>
 *
 * @author 吴宇亮 on 2019/2/25 0025 19:02
 */
public class _QrcodeUtil {

    /**
     * 二维码加密，将'文字'转换成'二维码'。以流式进行图片返回 <br/>
     * 参考博客：https://www.bilibili.com/video/av28985728/?p=4 <br/>
     *
     * @param qrcodeMsg 二维码的内容
     * @param width 二维码的宽
     * @param height 二维码的高
     * @param formatName 生成二维码的图片格式
     */
    public static void encoding(String qrcodeMsg, int width, int height, String formatName, HttpServletResponse response) throws IOException, WriterException {
        BufferedImage bufferedImage = getBufferedImage(qrcodeMsg, width, height);
        ServletOutputStream outputStream = response.getOutputStream();
        ImageIO.write(bufferedImage, formatName, outputStream);
    }

    /**
     * 二维码加密，将'文字'转换成'二维码'，以虚拟路径进行图片返回 <br/>
     * 参考博客：https://www.bilibili.com/video/av28985728/?p=4 <br/>
     *
     * @param qrcodeMsg 二维码的内容
     * @param width 二维码的宽
     * @param height 二维码的高
     * @param formatName 生成二维码的图片格式
     * @return 返回生成的文件的虚拟路径
     */
    public static String encoding(String qrcodeMsg, int width, int height, String formatName) throws IOException, WriterException {
        BufferedImage bufferedImage = getBufferedImage(qrcodeMsg, width, height);
        //将内存中的一张图片对象写到file文件里
        String prefix = _PropertyUtil.propertiesMap.get(_PropertyUtil.FRAMEWORK_PROPERTIES).getProperty("virtualPath");
        String suffix = "/" + System.currentTimeMillis() + "." + formatName;
        File file = new File(prefix + suffix);
        ImageIO.write(bufferedImage, formatName, file);
        //返回生成的路径
        return "virtualPath" + suffix;
    }

    /**
     * 生成二维码
     * @param qrcodeMsg 二维码携带的信息
     * @param width 二维码的宽
     * @param height 二维码的高
     * @return 返回生成的BufferedImage内存图片
     * @throws WriterException
     */
    private static BufferedImage getBufferedImage(String qrcodeMsg, int width, int height) throws WriterException {
        Map<EncodeHintType, Object> settingMap = new HashMap<>();
        settingMap.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H); //容错率 H>Q>M>L
        settingMap.put(EncodeHintType.CHARACTER_SET, "utf-8"); //编码
        settingMap.put(EncodeHintType.MARGIN, 1); //外边距
        BitMatrix bitMatrix = new MultiFormatWriter().encode(qrcodeMsg, BarcodeFormat.QR_CODE, width, height, settingMap);

        //生成一个内存图片对象, 宽是width, 高是height, 它由RGB颜色组成（即任何颜色）
        BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
        for (int x = 0; x < width; x++) {
            for (int y = 0; y < height; y++) {
                //给内存的图片对象, 在(x, y)这个坐标, 涂上color颜色
                bufferedImage.setRGB(x, y, bitMatrix.get(x, y) ? Color.BLACK.getRGB() : Color.WHITE.getRGB());
            }
        }

        //画logo，暂不考虑

        return bufferedImage;
    }

    /**
     * 解析二维码
     */
    public static String decoding(InputStream inputStream) throws IOException, NotFoundException {
        //将inputstream -》 内存中的一张图片
        BufferedImage image = ImageIO.read(inputStream);
        LuminanceSource luminanceSource = new BufferedImageLuminanceSource(image);
        Binarizer binarizer = new HybridBinarizer(luminanceSource);
        BinaryBitmap map = new BinaryBitmap(binarizer);

        //解析
        Result decode = new MultiFormatReader().decode(map);
        return decode.getText();
    }



}
